#pragma once

#include "../comm/util.hpp"
#include "../comm/log.hpp"

#include <iostream>
#include <string>

#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/stat.h>
#include <fcntl.h>

namespace ns_compiler
{
    using namespace ns_util;
    using namespace ns_log;

    class Compiler
    {
    public:
        Compiler(){}
        ~Compiler(){}
    public:
    //编译成功返回true,编译报错则返回false。
    //形参只是一个文件的名字，是没有后缀的
        static bool Compile(const std::string &file_name)
        {
            pid_t pid = fork();
            if(pid < 0)
            {
                //创建子进程失败
                LOG(ERROR) << "内部错误,创建子进程失败" << "\n";
                return false;
            }
            else if(pid == 0)
            {
                //子进程
                
                //1.创建编译错误临时文件
                umask(0);
                int _stderr = open(PathUtil::CompilerError(file_name).c_str(),O_CREAT | O_WRONLY,0644);
                if(_stderr < 0)
                {
                    //编译错误临时文件创建失败
                    LOG(WARNING) << "没有成功形成stderr文件" << "\n";
                    exit(1);
                }

                //2.重定向标准错误到_stderr
                dup2(_stderr,2);

                //3.程序替换,让子进程执行编译程序
                execlp("g++","g++","-o",PathUtil::Exe(file_name).c_str(),PathUtil::Src(file_name).c_str(),"-std=c++11","-D","COMPILER_ONLINE",nullptr);
                //程序替换成功后不可能执行后面的代码
                LOG(ERROR) << "启动编译器g++失败，可能是参数错误" << "\n";
                exit(2);
            }
            else
            {
                //父进程

                //1.等待子进程
                waitpid(pid,nullptr,0);

                //2.如果可执行文件存在
                if(FileUtil::IsFileExists(PathUtil::Exe(file_name).c_str()))
                {
                    LOG(INFO) << PathUtil::Src(file_name) << " 编译成功!" << "\n";
                    return true;
                }
            }

            //可执行文件生成失败
            LOG(ERROR) << "编译失败,没有形成可执行程序" << "\n";
            return false;
        }
    };
}
